package com.tanhua.sso.service;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.tanhua.sso.mapper.UserMapper;
import com.tanhua.sso.pojo.User;
import io.jsonwebtoken.*;
import lombok.Value;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.codec.digest.DigestUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.rocketmq.spring.core.RocketMQTemplate;
import org.joda.time.DateTime;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;

import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;

@Service
@Slf4j
public class loginService {
    @Autowired
    private RedisTemplate<String,String>redisTemplate;
    @Autowired
    private UserMapper mapper;

    private String secret="itcast";


    @Autowired
    private RocketMQTemplate rocketMQTemplate;
    /**
     *
     * @param phone 手机号
     * @param code 验证码
     */
    public String login(String phone, String code) {
        //首先校验验证码，然后去redis中获取信息
        String redisKey = "CHECK_CODE_" + phone;
        boolean isNew = false;
        String s = redisTemplate.opsForValue().get(redisKey);
        //判断验证码是否正确，如果正确，校验手机号，如果不正确返回一个null
        if (!StringUtils.equals(code,s)){
            return null;
        }
        //校验完成后，需要废弃验证码

        redisTemplate.delete(redisKey);
        //通过手机号去数据库中找用户，如果不存在，进行注册，如果存在，则返回token
        QueryWrapper<User>queryWrapper=new QueryWrapper<>();
        queryWrapper.eq("mobile",phone);
        //通过mapper中的查询一个的方法，进行判断
        User user=this.mapper.selectOne(queryWrapper);
        //如果为null就创建，如果不为null就返回token
        if (null==user){
            //需要注册该用户
            user = new User();
            user.setMobile(phone);
            user.setPassword(DigestUtils.md5Hex("123456"));

            //注册新用户
            this.mapper.insert(user);
            isNew = true;
        }

        //生成token
        Map<String, Object> claims = new HashMap<String, Object>();
        claims.put("id", user.getId());

        // 生成token
        String token = Jwts.builder()
                .setClaims(claims) //payload，存放数据的位置，不能放置敏感数据，如：密码等
                .signWith(SignatureAlgorithm.HS256, secret) //设置加密方法和加密盐
                .setExpiration(new DateTime().plusHours(12).toDate()) //设置过期时间，12小时后过期
                .compact();
        //返回token数据和true或者first
        return token+","+isNew;
    }

    public User queryUserByToken(String token) {
        try {
            //token解析
            Map<String,Object>body=Jwts.parser()
                    .setSigningKey(secret)
                    .parseClaimsJws(token)
                    .getBody();
            User user=new User();
            user.setId(Long.valueOf(body.get("id").toString()));

            //需要返回user对象中的mobile，需要查询数据库获取到mobile数据
            //如果每次都查询数据库，必然会导致性能问题，需要对用户的手机号进行缓存操作
            //数据缓存时，需要设置过期时间，过期时间要与token的时间一致
            //如果用户修改了手机号，需要同步修改redis中的数据

            //redis中mobile的id
            String redisKey = "TANHUA_USER_MOBILE_" + user.getId();
            //redis中存在，那么就进行赋值给user
            if (redisTemplate.hasKey(redisKey)){
                String mobile = redisTemplate.opsForValue().get(redisKey);
                user.setMobile(mobile);
            }else {
                //如果不存在就通过id查询，然后创建
                User user1 = mapper.selectById(user.getId());
                user.setMobile(user1.getMobile());

                long timeout = Long.valueOf(body.get("exp").toString())
                        * 1000 - System.currentTimeMillis();
                this.redisTemplate.opsForValue().set(redisKey, user1.getMobile(),
                        timeout, TimeUnit.MILLISECONDS);
            }
            return user;
        } catch (ExpiredJwtException e) {
            log.info("token已经过期！token="+token);
        } catch (Exception e) {
            log.error("token无效！token="+token,e);

        }
        return null;
    }
}
